import os


def create_video_list(video_dir, video_filename):
    video_names = os.listdir(video_dir)
    with  open(video_filename,'w',encoding="utf-8") as f:
        for item in video_names:
            f.writelines(item)
            f.write("\n")

if __name__ == '__main__':
    video_dir = r"/home/whut/D/video_data/train_fps6"
    video_filename = r"74_image_dirs.txt"
    create_video_list(video_dir, video_filename)
